Numpy function to help avoid the use of any loops or if-else statements